from langchain_core.output_parsers import StrOutputParser

from a0base.base_llm import pop_llm

# print(pop_llm.invoke("计算数字的加法运算：1+100=?"))

# 定义格式化
parser = StrOutputParser()

# 问答测试
llm_chain = pop_llm | parser  # 一行搞定“模型→解析”
answer = llm_chain.invoke("用一句话解释牛顿第二定律")
print(answer)  # 已经去掉 Message 外壳，纯字符串

# StrOutputParser 只做两件事：
# 把 AIMessage/ChatMessage 等对象转成纯字符串
# 默认 strip() 去掉首尾空白
